package cn.ctbu.xiaopeng.studentscoremanage.api;

import cn.ctbu.xiaopeng.studentscoremanage.enums.ResultEnum;
import cn.ctbu.xiaopeng.studentscoremanage.exception.ResultException;
import cn.ctbu.xiaopeng.studentscoremanage.pojo.dto.CourseQueryDTO;
import cn.ctbu.xiaopeng.studentscoremanage.pojo.entity.Course;
import cn.ctbu.xiaopeng.studentscoremanage.pojo.vo.CoursePageVO;
import cn.ctbu.xiaopeng.studentscoremanage.pojo.vo.Result;
import cn.ctbu.xiaopeng.studentscoremanage.pojo.dto.PageQueryDTO;
import cn.ctbu.xiaopeng.studentscoremanage.pojo.vo.page.PageResult;
import cn.ctbu.xiaopeng.studentscoremanage.service.CourseService;
import cn.ctbu.xiaopeng.studentscoremanage.utils.BuildTbSortUtils;
import cn.ctbu.xiaopeng.studentscoremanage.utils.PageResultUtils;
import cn.ctbu.xiaopeng.studentscoremanage.utils.ResultUtils;
import cn.hutool.core.bean.BeanUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.domain.*;
import org.springframework.web.bind.annotation.*;

import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * 课程信息接口
 *
 * @author xiaopeng
 * @version 1.0
 */
@RestController
@RequestMapping("/api/course")
@RequiredArgsConstructor
@Slf4j
public class CourseApiController {
    private final CourseService courseService;

    /**
     * 查询所有课程信息
     *
     * @return
     */
    @GetMapping("/getAll")
    public Result<List<Course>> getAllCourse() {
        return ResultUtils.success(courseService.getAllCourse());
    }

    /**
     * 查询所有课程种类名称
     * @return
     */
    @GetMapping("/getAllDifferentCName")
    public Result<List<Course>> getAllDifferentCName() {
        // 获取所有课程
        List<Course> allCourses = courseService.getAllCourse();

        // 使用 stream 通过课程名称去重，保留 Course 对象
        List<Course> distinctCourses = allCourses.stream()
                .filter(distinctByKey(Course::getCname))  // 通过课程名称去重
                .collect(Collectors.toList());  // 收集去重后的课程列表

        return ResultUtils.success(distinctCourses);
    }

    // 辅助方法：根据课程名称去重
    public static <T> java.util.function.Predicate<T> distinctByKey(
            java.util.function.Function<? super T, ?> keyExtractor) {
        java.util.Set<Object> seen = new java.util.HashSet<>();
        return t -> seen.add(keyExtractor.apply(t));
    }

    /**
     * 根据id查询课程信息
     *
     * @param id
     * @return
     */
    @GetMapping("/{id}")
    public Result<Course> findById(@PathVariable Integer id) {
        return ResultUtils.success(courseService.findById(id));
    }


    /**
     * 根据课程姓名模糊查询课程信息
     *
     * @param name
     * @return
     */
    @GetMapping("/query-name")
    public Result<List<Course>> findByNameStartWith(String name) {
        return ResultUtils.success(courseService.findByCnameStartingWith(name));
    }

    /**
     * 添加课程
     *
     * @param course
     * @return
     */
    @PostMapping("/add")
    public Result<Course> add(@RequestBody Course course) {
        if (course != null) {
            return ResultUtils.success(courseService.insert(course));
        } else {
            return ResultUtils.error("对象为空！");
        }
    }

    /**
     * 根据id删除课程
     *
     * @param id
     */
    @DeleteMapping("/delete/{id}")
    public Result delete(@PathVariable Integer id) {
        courseService.deleteById(id);
        return ResultUtils.success();
    }

    /**
     * 批量删除课程
     *
     * @param idsPath 路径中以逗号分隔的课程ID
     * @return 删除结果
     */
    @DeleteMapping("/delete/batch/{idsPath}")
    public Result deleteCourseByIds(@PathVariable String idsPath) {
        try {
            // 将路径中的逗号分隔字符串解析成 List<Integer>
            List<Integer> ids = Stream.of(idsPath.split(","))
                    .map(Integer::parseInt)
                    .collect(Collectors.toList());

            courseService.deleteCoursesByIds(ids);
            return ResultUtils.success("批量删除成功");
        } catch (IllegalArgumentException e) {
            return ResultUtils.error("删除失败：" + e.getMessage());
        } catch (Exception e) {
            return ResultUtils.error("系统异常：" + e.getMessage());
        }
    }

    /**
     * 修改课程信息
     *
     * @param course
     * @return
     */
    @PutMapping("/update")
    public Result update(@RequestBody Course course) {
        if (course != null) {
            return ResultUtils.success(courseService.update(course));
        } else {
            return ResultUtils.error("对象为空！");
        }
    }

    /**
     * 分页查询课程信息,可以实现按照表头排序
     *
     * @param pageQueryDTO
     * @return
     */
    @PostMapping("/page")
    /**
     * 根据页面查询课程信息，支持排序
     *
     * @param pageQueryVO 包含分页和排序信息的查询对象，如果为null，则使用默认分页和排序
     * @return 返回分页后的课程信息和总记录数
     */
    public PageResult getAllCourseByPageWithSort(@RequestBody PageQueryDTO<CourseQueryDTO> pageQueryDTO) {
        // 默认分页参数
        int pageIndex = 0;
        int pageSize = 10;

        if (pageQueryDTO == null) {
            // pageQueryVO为空直接走默认分页查询
            Pageable pageable = PageRequest.of(pageIndex, pageSize);
            Page<Course> courses = courseService.getAllWithPage(pageable); // 使用默认分页和排序
            return PageResultUtils.success(courses.getContent(), courses.getTotalElements());
        } else {
            // 根据传入的页码和页面大小初始化分页参数
            pageIndex = pageQueryDTO.getPage() > 0 ? pageQueryDTO.getPage() - 1 : 0;
            pageSize = pageQueryDTO.getLimit();

            // 初始化排序对象
            Sort sort = Sort.unsorted();
            if (pageQueryDTO.getSorts() != null && !pageQueryDTO.getSorts().isEmpty()) {
                // 构建Sort对象，根据前端传入的多个排序参数
                sort = BuildTbSortUtils.buildSort(pageQueryDTO.getSorts());
            } else {
                // 如果没有排序规则，则默认按照id升序排序
                sort = Sort.by(Sort.Direction.ASC, "id");
            }

            // 创建Pageable对象，包含分页和排序信息
            Pageable pageable = PageRequest.of(pageIndex, pageSize, sort);

            if (pageQueryDTO.getData() != null) {
                // 根据传入的课程信息进行查询
                CourseQueryDTO course = pageQueryDTO.getData();

                // 执行查询
                Page<CoursePageVO> coursePage = courseService.getCoursePageByConditions(course, pageable);
                List<CoursePageVO> coursePageContent = coursePage.getContent();

                // 返回分页结果
                return PageResultUtils.success(BeanUtil.copyToList(coursePageContent, CoursePageVO.class),
                        coursePage.getTotalElements());
            } else {
                // 如果pageQueryVO不为空，但data为空，则抛出异常
                throw new ResultException(ResultEnum.COMMON_ERROR);
            }
        }
    }

}
